package com.momo.hr.config;

import com.momo.hr.bean.MailConstants;
import com.momo.hr.service.MailSendLogService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.core.Binding;
import org.springframework.amqp.core.BindingBuilder;
import org.springframework.amqp.core.DirectExchange;
import org.springframework.amqp.core.Queue;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;


/**
 * @author 朱博
 * rabbit保证消息可靠性配置文件
 */
@Configuration
public class RabbitConfig {

    public final static Logger logger = LoggerFactory.getLogger(RabbitConfig.class);

    //这里为RabbitMQ的链接工厂（Spring中的RabbitTemplate中用的也是这个工厂来创建链接的 ）
    @Autowired
    CachingConnectionFactory cachingConnectionFactory;

    @Autowired
    MailSendLogService mailSendLogService;

    //我们来自定义rabbitTemplate
    @Bean
    RabbitTemplate rabbitTemplate(){
        //这里需要CachingConnectionFactory
        RabbitTemplate rabbitTemplate = new RabbitTemplate(cachingConnectionFactory);
        //设置发送消息后，并且RabbitMq收到消息后的返回
        /**
         * data:发送的数据
         * ack: 是否发送成功
         * cause:
         */
        rabbitTemplate.setConfirmCallback((data,ack,cause) -> {
                if(ack){
                    //消息发送成功
                    logger.info("消息Id"+data.getId()+"消息发送成功");
                    //更新数据库信息（1 表示消息投递成功）
                    mailSendLogService.updateMailSendLogStatus(data.getId(), 1);
                } else {
                    logger.info("消息发送失败");
                }
        });
        //交换机到路由的时候出错
        //也就是说没有准确的投递到对应的Queue中去就会报错
        rabbitTemplate.setReturnCallback((msg,repCode,repText,exchane,routeKey) -> {
            logger.info("交换机出错");
        });
        return rabbitTemplate;
    }

    /**
     * 指定队列
     * @return 指定队列名字
     */
    @Bean
    Queue mail(){
        //1、设置队列的名字2、指定为持久队列
        return new Queue(MailConstants.MAIL_QUEUE_NAME,true);
    }

    /**
     * 指定交换机
     * @return
     */
    @Bean
    DirectExchange mailDirectExchange(){
        //1、设置交换机名字 2、指定为持久队列 3、指定这个交换机不去自动删除
        return new DirectExchange(MailConstants.MAIL_EXCHANGE_NAME,true,false);
    }

    /**
     * 绑定交换机 和队列
     */
    @Bean
    Binding mailBinding(){
        //1、第一个为队列 后者是 交换机 路由的Key叫MAIL_ROUTING_KEY_NAME
        return BindingBuilder.bind(mail()).to(mailDirectExchange()).with(MailConstants.MAIL_ROUTING_KEY_NAME);
    }
}
